package com.rop.znyx.common.config;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Nonnull;
import javax.annotation.PostConstruct;

/**
 * Created with IntelliJ IDEA
 * User: Rop
 * Date: 2023/7/3
 * Time: 15:39
 * Description:
 */

@Slf4j
@Component
@RequiredArgsConstructor
public class MQProducerAckConfig implements RabbitTemplate.ReturnCallback, RabbitTemplate.ConfirmCallback {

    private final RabbitTemplate rabbitTemplate;

    /**
     * 设置 表示修饰一个非静态的void方法，在服务器加载Servlet的时候运行。并且只执行一次！
     */
    @PostConstruct
    public void init() {
        rabbitTemplate.setReturnCallback(this);
        rabbitTemplate.setConfirmCallback(this);
    }

    /**
     * 表示消息是否正确的发送到了交换机上
     *
     * @param correlationData 消息的载体
     * @param ack             判断是否发送到了交换机
     * @param cause           原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            System.out.println("消息发送成功");
            log.info("消息发送成功");
        } else {
            System.err.println("消息发送失败:{}" + cause);
            log.error("消息发送失败:{}", cause);
        }
    }

    /**
     * 消息如果没有正确发送到队列中，则会走这个方法！如果消息被正常处理，则这个方法不会走
     */
    @Override
    public void returnedMessage(@Nonnull Message message, int replyCode, @Nonnull String replyText, @Nonnull String exchange, @Nonnull String routingKey) {
        System.err.println("消息主体: " + new String(message.getBody()));
        System.err.println("应答码: " + replyCode);
        System.err.println("描述：" + replyText);
        System.err.println("消息使用的交换器 exchange : " + exchange);
        System.err.println("消息使用的路由键 routing : " + routingKey);
        log.error("消息主体: {}", new String(message.getBody()));
        log.error("应答码: {}", replyCode);
        log.error("描述：{}", replyText);
        log.error("消息使用的交换器 exchange : {}", exchange);
        log.error("消息使用的路由键 routing : {}", routingKey);
    }
}
